home *** CD-ROM | disk | FTP | other *** search
/ Nebula 1 / Nebula One.iso / Mail / pine3.92 / pine / osdep / pipe < prev    next >
Encoding:
Text File  |  1996-03-05  |  10.3 KB  |  381 lines
  1. /*======================================================================
  2.     pipe
  3.     
  4.     Initiate I/O to and from a process.  These functions are similar to 
  5.     popen and pclose, but both an incoming stream and an output file are 
  6.     provided.
  7.    
  8.  ====*/
  9.  
  10. #ifndef    STDIN_FILENO
  11. #define    STDIN_FILENO    0
  12. #endif
  13. #ifndef    STDOUT_FILENO
  14. #define    STDOUT_FILENO    1
  15. #endif
  16. #ifndef    STDERR_FILENO
  17. #define    STDERR_FILENO    2
  18. #endif
  19.  
  20.  
  21. /*
  22.  * Defs to help fish child's exit status out of wait(2)
  23.  */
  24. #ifdef    HAVE_WAIT_UNION
  25. #define WaitType    union wait
  26. #ifndef    WIFEXITED
  27. #define    WIFEXITED(X)    (!(X).w_termsig)    /* child exit by choice */
  28. #endif
  29. #ifndef    WEXITSTATUS
  30. #define    WEXITSTATUS(X)    (X).w_retcode        /* childs chosen exit value */
  31. #endif
  32. #else
  33. #define    WaitType    int
  34. #ifndef    WIFEXITED
  35. #define    WIFEXITED(X)    (!((X) & 0xff))        /* low bits tell how it died */
  36. #endif
  37. #ifndef    WEXITSTATUS
  38. #define    WEXITSTATUS(X)    (((X) >> 8) & 0xff)    /* high bits tell exit value */
  39. #endif
  40. #endif
  41.  
  42.  
  43. /*
  44.  * Global's to helpsignal handler tell us child's status has changed...
  45.  */
  46. short    child_signalled;
  47. short    child_jump = 0;
  48. jmp_buf child_state;
  49.  
  50.  
  51.  
  52. /*----------------------------------------------------------------------
  53.      Spawn a child process and optionally connect read/write pipes to it
  54.  
  55.   Args: command -- string to hand the shell
  56.     outfile -- address of pointer containing file to receive output
  57.     errfile -- address of pointer containing file to receive error output
  58.     mode -- mode for type of shell, signal protection etc...
  59.   Returns: pointer to alloc'd PIPE_S on success, NULL otherwise
  60.  
  61.   The outfile is either NULL, a pointer to a NULL value, or a pointer
  62.   to the requested name for the output file.  In the pointer-to-NULL case
  63.   the caller doesn't care about the name, but wants to see the pipe's
  64.   results so we make one up.  It's up to the programmer to make sure
  65.   the free storage containing the name is cleaned up.
  66.  
  67.   Mode bits serve several purposes.
  68.     PIPE_WRITE tells us we need to open a pipe to write the child's
  69.     stdin.
  70.     PIPE_READ tells us we need to open a pipe to read from the child's
  71.     stdout/stderr.  *NOTE*  Having neither of the above set means 
  72.     we're not setting up any pipes, just forking the child and exec'ing
  73.     the command.  Also, this takes precedence over any named outfile.
  74.     PIPE_STDERR means we're to tie the childs stderr to the same place
  75.     stdout is going.  *NOTE* This only makes sense then if PIPE_READ
  76.     or an outfile is provided.  Also, this takes precedence over any
  77.     named errfile.
  78.     PIPE_PROT means to protect the child from the usual nasty signals
  79.     that might cause premature death.  Otherwise, the default signals are
  80.     set so the child can deal with the nasty signals in its own way.     
  81.     PIPE_NOSHELL means we're to exec the command without the aid of
  82.     a system shell.  *NOTE* This negates the affect of PIPE_USER.
  83.     PIPE_USER means we're to try executing the command in the user's
  84.     shell.  Right now we only look in the environment, but that may get
  85.     more sophisticated later.
  86.     PIPE_RESET means we reset the terminal mode to what it was before
  87.     we started pine and then exec the command.
  88.  ----*/
  89. PIPE_S *
  90. open_system_pipe(command, outfile, errfile, mode)
  91.     char  *command;
  92.     char **outfile, **errfile;
  93.     int    mode;
  94. {
  95.     PIPE_S *syspipe = NULL;
  96.     char    shellpath[32], *shell;
  97.     int     p[2], oparentd = -1, ochildd = -1, iparentd = -1, ichildd = -1;
  98.  
  99.     dprint(5, (debugfile, "Opening pipe: \"%s\" (%s%s%s%s%s%s)\n",command,
  100.            (mode & PIPE_WRITE)   ? "W":"", (mode & PIPE_READ)  ? "R":"",
  101.            (mode & PIPE_NOSHELL) ? "N":"", (mode & PIPE_PROT)  ? "P":"",
  102.            (mode & PIPE_USER)    ? "U":"", (mode & PIPE_RESET) ? "T":""));
  103.  
  104.     if(!(mode & PIPE_READ)){
  105.     if(outfile && !*outfile)
  106.       *outfile = temp_nam(NULL, "pine_p");    /* asked for, but not named? */
  107.  
  108.     if(errfile && !*errfile)
  109.       *errfile = temp_nam(NULL, "pine_p");    /* ditto */
  110.     }
  111.  
  112.     if(mode & (PIPE_WRITE | PIPE_READ)){
  113.     if(mode & PIPE_WRITE){
  114.         pipe(p);                /* alloc pipe to write child */
  115.         oparentd = p[STDOUT_FILENO];
  116.         ichildd  = p[STDIN_FILENO];
  117.     }
  118.  
  119.     if(mode & PIPE_READ){
  120.         pipe(p);                /* alloc pipe to read child */
  121.         iparentd = p[STDIN_FILENO];
  122.         ochildd  = p[STDOUT_FILENO];
  123.     }
  124.     }
  125.     else{
  126.     flush_status_messages(0);        /* just clean up display */
  127.     ClearScreen();
  128.     fflush(stdout);
  129.     }
  130.  
  131.     syspipe = (PIPE_S *)fs_get(sizeof(PIPE_S));
  132.     memset(syspipe, 0, sizeof(PIPE_S));
  133.  
  134.     if((syspipe->mode = mode) & PIPE_RESET)
  135.       Raw(0);
  136.  
  137. #ifdef    SIGCHLD
  138.     /*
  139.      * Prepare for demise of child.  Use SIGCHLD if it's available so
  140.      * we can do useful things, like keep the IMAP stream alive, while
  141.      * we're waiting on the child.
  142.      */
  143.     child_signalled = child_jump = 0;
  144.     (void)signal(SIGCHLD,  child_signal);
  145. #endif
  146.  
  147.     if((syspipe->pid = vfork()) == 0){
  148.      /* reset child's handlers in requested fashion... */
  149.     (void)signal(SIGINT,  (mode & PIPE_PROT) ? SIG_IGN : SIG_DFL);
  150.     (void)signal(SIGQUIT, (mode & PIPE_PROT) ? SIG_IGN : SIG_DFL);
  151.     (void)signal(SIGHUP,  (mode & PIPE_PROT) ? SIG_IGN : SIG_DFL);
  152. #ifdef    SIGCHLD
  153.     (void) signal(SIGCHLD,  SIG_DFL);
  154. #endif
  155.  
  156.     /* if parent isn't reading, and we have a filename to write */
  157.     if(!(mode & PIPE_READ) && outfile){    /* connect output to file */
  158.         int output = creat(*outfile, 0600);
  159.         dup2(output, STDOUT_FILENO);
  160.         if(mode & PIPE_STDERR)
  161.           dup2(output, STDERR_FILENO);
  162.         else if(errfile)
  163.           dup2(creat(*errfile, 0600), STDERR_FILENO);
  164.     }
  165.  
  166.     if(mode & PIPE_WRITE){            /* connect process input */
  167.         close(oparentd);
  168.         dup2(ichildd, STDIN_FILENO);    /* tie stdin to pipe */
  169.         close(ichildd);
  170.     }
  171.  
  172.     if(mode & PIPE_READ){            /* connect process output */
  173.         close(iparentd);
  174.         dup2(ochildd, STDOUT_FILENO);    /* tie std{out,err} to pipe */
  175.         if(mode & PIPE_STDERR)
  176.           dup2(ochildd, STDERR_FILENO);
  177.         else if(errfile)
  178.           dup2(creat(*errfile, 0600), STDERR_FILENO);
  179.  
  180.         close(ochildd);
  181.     }
  182.  
  183.     if(mode & PIPE_NOSHELL){
  184.         char   **argv, **ap, *p, *cmd;
  185.         size_t   n;
  186.  
  187.         /* parse the arguments into a list */
  188.         for(cmd = cpystr(command); *cmd && isspace(*cmd); cmd++)
  189.           ;                    /* swallow leading ws */
  190.  
  191.         for(p = cmd, n = 2; *p; p++)    /* count the args */
  192.           if(isspace(*p) && *(p+1) && !isspace(*(p+1)))
  193.         n++;
  194.  
  195.         argv = ap = (char **)fs_get(n * sizeof(char *));
  196.         memset(argv, 0, n * sizeof(char *));
  197.         for(p = cmd; *p && !isspace(*p); p++)
  198.           ;
  199.  
  200.         if(*p)                /* tie off command name */
  201.           *p++ = '\0';
  202.  
  203.         *ap++ = cpystr(cmd);
  204.         while(*p){                /* collect args */
  205.         while(*p && isspace(*p))
  206.           *p++ = '\0';
  207.  
  208.         *ap++ = (*p) ? p : NULL;
  209.         while(*p && !isspace(*p))
  210.           p++;
  211.         }
  212.  
  213.         execvp(cmd, argv);
  214.     }
  215.     else{
  216.         if(mode & PIPE_USER){
  217.         char *env, *sh;
  218.         if((env = getenv("SHELL")) && (sh = strrchr(env, '/'))){
  219.             shell = sh + 1;
  220.             strcpy(shellpath, env);
  221.         }
  222.         else{
  223.             shell = "csh";
  224.             strcpy(shellpath, "/bin/csh");
  225.         }
  226.         }
  227.         else{
  228.         shell = "sh";
  229.         strcpy(shellpath, "/bin/sh");
  230.         }
  231.  
  232.         execl(shellpath, shell, command ? "-c" : 0, command, 0);
  233.     }
  234.  
  235.     fprintf(stderr, "Can't exec %s\nReason: %s",
  236.         command, error_description(errno));
  237.     _exit(-1);
  238.     }
  239.  
  240.     if(syspipe->pid > 0){
  241.     syspipe->isig = signal(SIGINT,  SIG_IGN); /* Reset handlers to make */
  242.     syspipe->qsig = signal(SIGQUIT, SIG_IGN); /* sure we don't come to  */
  243.     syspipe->hsig = signal(SIGHUP,  SIG_IGN); /* a premature end...     */
  244.  
  245.     if(mode & PIPE_WRITE){
  246.         close(ichildd);
  247.         syspipe->ofilep = fdopen(oparentd, "w");
  248.     }
  249.  
  250.     if(mode & PIPE_READ){
  251.         close(ochildd);
  252.         syspipe->ifilep = fdopen(iparentd, "r");
  253.     }
  254.  
  255.     dprint(5, (debugfile, "PID: %d, COMMAND: %s\n",syspipe->pid,command));
  256.     }
  257.     else{
  258.     if(mode & (PIPE_WRITE | PIPE_READ)){
  259.         if(mode & PIPE_WRITE){
  260.         close(oparentd);
  261.         close(ichildd);
  262.         }
  263.  
  264.         if(mode & PIPE_READ){
  265.         close(iparentd);
  266.         close(ochildd);
  267.         }
  268.     }
  269.     else{
  270.         ClearScreen();
  271.         ps_global->mangled_screen = 1;
  272.     }
  273.  
  274.     if(mode & PIPE_RESET)
  275.       Raw(1);
  276.  
  277. #ifdef    SIGCHLD
  278.     (void) signal(SIGCHLD,  SIG_DFL);
  279. #endif
  280.  
  281.     q_status_message1(SM_ORDER,3,3, "Error executing external command: %s",
  282.               error_description(errno));
  283.     fs_give((void **)&syspipe);
  284.     if(outfile)
  285.       fs_give((void **)outfile);
  286.  
  287.     dprint(1, (debugfile, "CAN'T FORK FOR COMMAND: %s\n", command));
  288.     }
  289.  
  290.     return(syspipe);
  291. }
  292.  
  293.  
  294.  
  295. /*----------------------------------------------------------------------
  296.     Close pipe previously allocated and wait for child's death
  297.  
  298.   Args: syspipe -- address of pointer to struct returned by open_system_pipe
  299.   Returns: returns exit status of child or -1 if invalid syspipe
  300.  ----*/
  301. int
  302. close_system_pipe(syspipe)
  303.     PIPE_S **syspipe;
  304. {
  305.     WaitType stat;
  306.     int         status;
  307.  
  308.     if(!syspipe || !*syspipe)
  309.       return(-1);
  310.  
  311.     if((*syspipe)->ofilep)
  312.       fclose((*syspipe)->ofilep);
  313.  
  314.     if((*syspipe)->ifilep)
  315.       fclose((*syspipe)->ifilep);
  316.  
  317. #ifdef    SIGCHLD
  318.     {
  319.     SigType (*alarm_sig)();
  320.     int    old_cue = F_ON(F_SHOW_DELAY_CUE, ps_global);
  321.  
  322.     /*
  323.      * remember the current SIGALRM handler, and make sure it's
  324.      * installed when we're finished just in case the longjmp
  325.      * out of the SIGCHLD handler caused sleep() to lose it.
  326.      * Don't pay any attention to that man behind the curtain.
  327.      */
  328.     alarm_sig = signal(SIGALRM, SIG_IGN);
  329.     (void) signal(SIGALRM, alarm_sig);
  330.     F_SET(F_SHOW_DELAY_CUE, ps_global, 0);
  331.     ps_global->noshow_timeout = 1;
  332.     while(!child_signalled){
  333.         new_mail(0, 2, 0);        /* wake up and prod server */
  334.  
  335.         if(!child_signalled){
  336.         if(setjmp(child_state) == 0){
  337.             child_jump = 1;    /* prepare to wake up */
  338.             sleep(600);        /* give it 5mins to happend */
  339.         }
  340. #ifdef POSIX_SIGNALS
  341.         else
  342.           sigrelse(SIGCHLD);    /* unblock signal after longjmp */
  343. #endif
  344.         }
  345.  
  346.         child_jump = 0;
  347.     }
  348.  
  349.     ps_global->noshow_timeout = 0;
  350.     F_SET(F_SHOW_DELAY_CUE, ps_global, old_cue);
  351.     (void) signal(SIGALRM, alarm_sig);
  352.     (void) signal(SIGCHLD,  SIG_DFL);
  353.     }
  354. #endif
  355.  
  356.     /*
  357.      * Call c-client's pid reaper to wait() on the demise of our child,
  358.      * then fish out its exit status...
  359.      */
  360.     grim_pid_reap_status((*syspipe)->pid, 0, &stat);
  361.     status = WIFEXITED(stat) ? WEXITSTATUS(stat) : -1;
  362.  
  363.     /*
  364.      * restore original handlers...
  365.      */
  366.     (void)signal(SIGINT,  (*syspipe)->isig);
  367.     (void)signal(SIGHUP,  (*syspipe)->hsig);
  368.     (void)signal(SIGQUIT, (*syspipe)->qsig);
  369.  
  370.     if((*syspipe)->mode & PIPE_RESET)        /* restore our tty modes */
  371.       Raw(1);
  372.  
  373.     if(!((*syspipe)->mode & (PIPE_WRITE | PIPE_READ))){
  374.     ClearScreen();                /* No I/O to forked child */
  375.     ps_global->mangled_screen = 1;
  376.     }
  377.  
  378.     fs_give((void **)syspipe);
  379.     return(status);
  380. }
  381.